# !/usr/bin/env python
# -- coding: utf-8 --
# @Author zengxiaohui
# Datatime:8/9/2021 8:35 AM
# @File:small_model
import torch
path  = "/home/zengxh/workspace/lcnn/logs/210809-215452-88f281a-baseline/npz/000360000/checkpoint.pth"
model = torch.load(path)
torch.save(
    {
        "arch": model.__class__.__name__,
        "model_state_dict": model['model_state_dict'],
    },
    'creepageDistance22.pth',
)